home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Resources / Chat & Communication / Digsby build 37 / digsby_setup.exe / lib / PIL / IcnsImagePlugin.pyo (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2008-10-13  |  5.5 KB  |  191 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.5)
  3.  
  4. import Image
  5. import ImageFile
  6. import string
  7. import struct
  8. HEADERSIZE = 8
  9.  
  10. def nextheader(fobj):
  11.     return struct.unpack('>4sI', fobj.read(HEADERSIZE))
  12.  
  13.  
  14. def read_32t(fobj, .1, .2):
  15.     (start, length) = .1
  16.     (width, height) = .2
  17.     fobj.seek(start)
  18.     sig = fobj.read(4)
  19.     if sig != '\x00\x00\x00\x00':
  20.         raise SyntaxError, 'Unknown signature, expecting 0x00000000'
  21.     
  22.     return read_32(fobj, (start + 4, length - 4), (width, height))
  23.  
  24.  
  25. def read_32(fobj, .1, size):
  26.     (start, length) = .1
  27.     fobj.seek(start)
  28.     sizesq = size[0] * size[1]
  29.     if length == sizesq * 3:
  30.         indata = fobj.read(length)
  31.         im = Image.frombuffer('RGB', size, indata, 'raw', 'RGB', 0, 1)
  32.     else:
  33.         im = Image.new('RGB', size, None)
  34.         for band_ix in range(3):
  35.             data = []
  36.             bytesleft = sizesq
  37.             while bytesleft > 0:
  38.                 byte = fobj.read(1)
  39.                 if not byte:
  40.                     break
  41.                 
  42.                 byte = ord(byte)
  43.                 if byte & 128:
  44.                     blocksize = byte - 125
  45.                     byte = fobj.read(1)
  46.                     for i in range(blocksize):
  47.                         data.append(byte)
  48.                     
  49.                 else:
  50.                     blocksize = byte + 1
  51.                     data.append(fobj.read(blocksize))
  52.                 bytesleft = bytesleft - blocksize
  53.                 if bytesleft <= 0:
  54.                     break
  55.                     continue
  56.             if bytesleft != 0:
  57.                 raise SyntaxError('Error reading %r channel [%r]' % (channel, bytesleft))
  58.             
  59.             band = Image.frombuffer('L', size, string.join(data, ''), 'raw', 'L', 0, 1)
  60.             im.im.putband(band.im, band_ix)
  61.         
  62.     return {
  63.         'RGB': im }
  64.  
  65.  
  66. def read_mk(fobj, .1, size):
  67.     (start, length) = .1
  68.     fobj.seek(start)
  69.     band = Image.frombuffer('L', size, fobj.read(size[0] * size[1]), 'raw', 'L', 0, 1)
  70.     return {
  71.         'A': band }
  72.  
  73.  
  74. class IcnsFile:
  75.     SIZES = {
  76.         (128, 128): [
  77.             ('it32', read_32t),
  78.             ('t8mk', read_mk)],
  79.         (48, 48): [
  80.             ('ih32', read_32),
  81.             ('h8mk', read_mk)],
  82.         (32, 32): [
  83.             ('il32', read_32),
  84.             ('l8mk', read_mk)],
  85.         (16, 16): [
  86.             ('is32', read_32),
  87.             ('s8mk', read_mk)] }
  88.     
  89.     def __init__(self, fobj):
  90.         self.dct = dct = { }
  91.         self.fobj = fobj
  92.         (sig, filesize) = nextheader(fobj)
  93.         if sig != 'icns':
  94.             raise SyntaxError, 'not an icns file'
  95.         
  96.         i = HEADERSIZE
  97.         while i < filesize:
  98.             (sig, blocksize) = nextheader(fobj)
  99.             i = i + HEADERSIZE
  100.             blocksize = blocksize - HEADERSIZE
  101.             dct[sig] = (i, blocksize)
  102.             fobj.seek(blocksize, 1)
  103.             i = i + blocksize
  104.  
  105.     
  106.     def itersizes(self):
  107.         sizes = []
  108.         for size, fmts in self.SIZES.items():
  109.             for fmt, reader in fmts:
  110.                 if self.dct.has_key(fmt):
  111.                     sizes.append(size)
  112.                     break
  113.                     continue
  114.             
  115.         
  116.         return sizes
  117.  
  118.     
  119.     def bestsize(self):
  120.         sizes = self.itersizes()
  121.         if not sizes:
  122.             raise SyntaxError, 'No 32bit icon resources found'
  123.         
  124.         return max(sizes)
  125.  
  126.     
  127.     def dataforsize(self, size):
  128.         dct = { }
  129.         for code, reader in self.SIZES[size]:
  130.             desc = self.dct.get(code)
  131.             if desc is not None:
  132.                 dct.update(reader(self.fobj, desc, size))
  133.                 continue
  134.         
  135.         return dct
  136.  
  137.     
  138.     def getimage(self, size = None):
  139.         if size is None:
  140.             size = self.bestsize()
  141.         
  142.         channels = self.dataforsize(size)
  143.         im = channels.get('RGB').copy()
  144.         
  145.         try:
  146.             im.putalpha(channels['A'])
  147.         except KeyError:
  148.             pass
  149.  
  150.         return im
  151.  
  152.  
  153.  
  154. class IcnsImageFile(ImageFile.ImageFile):
  155.     format = 'ICNS'
  156.     format_description = 'Mac OS icns resource'
  157.     
  158.     def _open(self):
  159.         self.icns = IcnsFile(self.fp)
  160.         self.mode = 'RGBA'
  161.         self.size = self.icns.bestsize()
  162.         self.info['sizes'] = self.icns.itersizes()
  163.         self.tile = ('',)
  164.  
  165.     
  166.     def load(self):
  167.         Image.Image.load(self)
  168.         if not self.tile:
  169.             return None
  170.         
  171.         self.load_prepare()
  172.         im = self.icns.getimage(self.size)
  173.         self.im = im.im
  174.         self.mode = im.mode
  175.         self.size = im.size
  176.         self.fp = None
  177.         self.icns = None
  178.         self.tile = ()
  179.         self.load_end()
  180.  
  181.  
  182. Image.register_open('ICNS', IcnsImageFile, (lambda x: x[:4] == 'icns'))
  183. Image.register_extension('ICNS', '.icns')
  184. if __name__ == '__main__':
  185.     import os
  186.     import sys
  187.     im = Image.open(open(sys.argv[1], 'rb'))
  188.     im.save('out.png')
  189.     os.startfile('out.png')
  190.  
  191.